38. CIFAR 分类示例
23 Cifar 分类 V1
model.eval()
上述代码丢失了 model.eval()
。
model.eval(
会将模型里的所有层级设置为评估模式。这样会影响到丢弃层等层级,但会允许所有节点接受评估。丢弃层是指在训练期间按照某个概率关闭节点的层级。在测试或验证模型之前,应该将模型设为评估模式,并且仅在训练循环期间将模型设为 model.train()
(训练模式)。
这一点体现在了之前的 notebook 代码和此 Github 代码库中。